#include "RingQueue.hpp"
#include "Task.hpp"
#include <iostream>
#include <cstdlib>
#include <ctime>
#include <pthread.h>
#include <sys/types.h>
#include <unistd.h>

std::string SelfName()
{
    char name[64];
    snprintf(name, sizeof(name), "thread[0x%0x]", pthread_self());
    return name;
}

void *ProducerRoutine(void *rq)
{
    // RingQueue<int> *ringqueue = static_cast<RingQueue<int> *>(rq);
    RingQueue<Task> *ringqueue = static_cast<RingQueue<Task> *>(rq);

    while (true)
    {
        // version1
        // int data = rand() % 10 + 1;
        // ringqueue->push(data);
        // std::cout << "生产完成，生产的数据是：" << data << std::endl; // 和阻塞队列不同的是，如果生产和消费的不是同一个位置，生产和消费是可以同时并发执行的。
        // sleep(1);

        // version2
        // 构建or获取任务
        int x = rand() % 100 + 1;
        int y = rand() % 10;
        char op = oper[rand() % oper.size()];
        // 生产任务
        Task t(x, y, op, mymath);
        ringqueue->Push(t);
        // 输出提示
        std::cout << SelfName() << "生产者派发了一个任务：" << t.toTaskString() << std::endl;

        sleep(1);
    }
}
void *ConsumerRoutine(void *rq)
{
    // RingQueue<int> *ringqueue = static_cast<RingQueue<int> *>(rq);
    RingQueue<Task> *ringqueue = static_cast<RingQueue<Task> *>(rq);

    while (true)
    {
        // version1
        // int data;
        // ringqueue->pop(&data);
        // std::cout << "消费完成，消费的数据是：" << data << std::endl;

        // version2
        Task t;
        ringqueue->Pop(&t);
        std::string result = t();
        std::cout << SelfName() << "消费者消费了一个任务：" << result << std::endl;
        //sleep(1);
    }
}

int main()
{
    srand((unsigned int)time(nullptr) ^ getpid());
    // RingQueue<int> *rq = new RingQueue<int>();
    RingQueue<Task> *rq = new RingQueue<Task>();

    // 单生产者、单消费者
    // pthread_t p, c;

    // pthread_create(&p, nullptr, ProducerRoutine, rq);
    // pthread_create(&p, nullptr, ConsumerRoutine, rq);

    // pthread_join(p, nullptr);
    // pthread_join(c, nullptr);

    // 多生产者、多消费者
    pthread_t p[4], c[8];

    for (int i = 0; i < 4; i++)
    {
        pthread_create(p + i, nullptr, ProducerRoutine, rq);
    }
    for (int i = 0; i < 8; i++)
    {
        pthread_create(c + i, nullptr, ConsumerRoutine, rq);
    }

    for (int i = 0; i < 4; i++)
    {
        pthread_join(p[i], nullptr);
    }
    for (int i = 0; i < 8; i++)
    {
        pthread_join(c[i], nullptr);
    }

    return 0;
}